#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
简化版合并脚本
将 res 文件夹中的所有 CSV 文件合并为一个 XLSX 文件
"""

import pandas as pd
import os
import glob
from datetime import datetime

def simple_merge():
    """简单的合并功能"""
    # 获取所有 CSV 文件
    csv_files = glob.glob("./res/*.csv")
    
    if not csv_files:
        print("没有找到 CSV 文件")
        return
    
    # 读取并合并
    all_dfs = []
    for file in sorted(csv_files):
        df = pd.read_csv(file)
        all_dfs.append(df)
        print(f"读取: {os.path.basename(file)} - {len(df)} 行")
    
    # 合并
    merged = pd.concat(all_dfs, ignore_index=True)
    
    # 保存
    output = f"merged_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
    merged.to_excel(output, index=False)
    
    print(f"\n合并完成: {output}")
    print(f"总行数: {len(merged)}")
    print(f"总列数: {len(merged.columns)}")

if __name__ == "__main__":
    simple_merge()
